/*
 * Copyright 2009 Original Author(s)
 * 
 * This file is part of Kommando
 * 
 * Kommando is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Kommando is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with Kommando.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.kommando.core.hotkey;

import java.util.HashMap;
import java.util.Map;

import org.kommando.core.command.Command;

/**
 * @author Peter De Bruycker
 */
public class CommandBindingRegistry {

	private Map<Command, Hotkey> bindings = new HashMap<Command, Hotkey>();
	private HotkeyService hotkeyService;

	public CommandBindingRegistry(HotkeyService hotkeyService) {
		this.hotkeyService = hotkeyService;
	}

	public Hotkey getBinding(Command command) {
		return bindings.get(command);
	}

	public Command getCommand(Hotkey key) {
		for (Map.Entry<Command, Hotkey> entry : bindings.entrySet()) {
			if (entry.getValue().equals(key)) {
				return entry.getKey();
			}
		}
		return null;
	}

	public void bind(Command command, Hotkey key) {
		if (bindings.containsKey(command)) {
			unbind(command);
		}

		bindings.put(command, key);
		hotkeyService.registerHotkey(key);
	}

	public void unbind(Command command) {
		hotkeyService.unregisterHotkey(bindings.get(command));
		bindings.remove(command);
	}
}
